package DP;

/**
 * 474.一和零  （给定背包容量,装满背包最多用多少个物品、二维背包容量）
 * 给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
 * 请你找出并返回 strs 的最大子集的长度，该子集中 最多 有 m 个 0 和 n 个 1 。
 * 如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。
 * dp[i][j]：最多有i个0和j个1的strs的最大子集的大小为dp[i][j]
 * 不取              取当前strs
 * dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
 */
public class findMaxForm {
    /**
     * 一刷
     */
    public int findMaxForm(String[] strs, int m, int n) {

        int[][] dp = new int[m + 1][n + 1];
        dp[0][0] = 0;
        for (String str : strs) {   //物品
            int x = 0;
            int y = 0;
            for (char c : str.toCharArray()) {
                if (c == '0') x++;
                else y++;
            }
            for (int i = m; i >= x; i--) {     //背包
                for (int j = n; j >= y; j--) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - x][j - y] + 1);
                }
            }
        }
        return dp[m][n];
    }

    /**
     * 二刷
     */
    public int findMaxForm2(String[] strs, int m, int n) {
        //dp[i][j]表示i个0和j个1时的最大子集
        int[][] dp = new int[m + 1][n + 1];
        int oneNum, zeroNum;
        for (String str : strs) {
            oneNum = 0;
            zeroNum = 0;
            for (char ch : str.toCharArray()) {
                if (ch == '0') {
                    zeroNum++;
                } else {
                    oneNum++;
                }
            }
            //相当于01背包的一维dp：倒序遍历
            for (int i = m; i >= zeroNum; i--) {
                for (int j = n; j >= oneNum; j--) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
                }
            }
        }
        return dp[m][n];
    }
}
